In [ ]:

    

import openni as opi
import numpy as np
import Image
import matplotlib.pyplot as plt
import pygame
from pygame.locals import KEYDOWN, K_ESCAPE
from pygame.color import THECOLORS


    

In [ ]:

    

ctx = opi.Context()
ctx.init()


    

In [ ]:

    

image_generator = opi.ImageGenerator()
image_generator.create(ctx)


    

In [ ]:

    

ctx.start_generating_all()


    

In [ ]:

    

def capture_rgb():
    rgb_frame = np.fromstring(image_generator.get_raw_image_map_bgr(),
                              dtype=np.uint8).reshape(480, 640, 3)
    
    im = Image.fromarray(rgb_frame)
    b, g, r = im.split()
    im = Image.merge("RGB", (r, g, b))
    #return np.array(im)
    return pygame.image.frombuffer(im.tostring(), im.size, 'RGB')


    

In [ ]:

    

plt.imshow(capture_rgb())


    

In [ ]:

    

#gentle remainder .. change the return value of capture_rgb
pygame.init()
running = True
screen = pygame.display.set_mode((640, 480), pygame.HWSURFACE | pygame.DOUBLEBUF)
pygame.display.set_caption('My First Camera')

while running:
    for event in pygame.event.get():
        if event.type == KEYDOWN and event.key == K_ESCAPE: running = False
    ctx.wait_any_update_all()
    screen.blit(capture_rgb(), (0, 0))
    pygame.display.update()
    pygame.display.flip()
    
pygame.display.quit()
pygame.quit()


    

In [ ]:

    

ctx.stop_generating_all()
ctx.shutdown()


    

In [ ]:

    

ctx = opi.Context()
ctx.init()


    

In [ ]:

    

depth_generator = opi.DepthGenerator()
depth_generator.create(ctx)
depth_generator.set_resolution_preset(opi.RES_VGA)
depth_generator.fps = 30


    

In [ ]:

    

ctx.start_generating_all()


    

In [ ]:

    

def capture_depth():
    depth_map = np.asarray(depth_generator.get_tuple_depth_map())
    depth_map = depth_map.reshape((480, 640))
    
    return depth_map


    

In [ ]:

    

plt.imshow(capture_depth())


    

In [ ]:

    

ctx.stop_generating_all()
ctx.shutdown()


    

In [ ]:

    

ctx = opi.Context()
ctx.init()


    

In [ ]:

    

ir_generator = opi.IRGenerator()
ir_generator.create(ctx)
ir_generator.set_resolution_preset(opi.RES_VGA)
ir_generator.fps = 30


    

In [ ]:

    

ctx.start_generating_all()


    

In [ ]:

    

def capture_IR():
    ir_map = np.asarray(ir_generator.get_tuple_ir_map())
    ir_map = ir_map.reshape((480, 640))
    
    return ir_map


    

In [ ]:

    

plt.imshow(capture_IR())


    

In [ ]:

    

ctx.stop_generating_all()
ctx.shutdown()


    

In [ ]: